#include <string>

#include <http_client.h>

namespace tc = triton::client;

class TextEmbedder {
public:
    TextEmbedder();
    std::vector<float> embedText(const std::string &text);
    std::vector<int64_t> tokenize(const std::string &text);

private:

private:
    std::string modelName_ = "text_embedding";
    std::string modelVersion_ = "";

    std::string tokenizerModelName_ = "transformers_tokenizer";
    std::string tokenizerModelVersion_ = "";

    std::string httpServerUrl_ = "http://localhost:8000";
    std::unique_ptr<tc::InferenceServerHttpClient> httpClient_;

};